//===----------------------------------------------------------------------===//
//
// Copyright (C) 2022 Sophgo Technologies Inc.  All rights reserved.
//
// SOPHON-DEMO is licensed under the 2-Clause BSD License except for the
// third-party components.
//
//===----------------------------------------------------------------------===//

#ifndef SOPHON_STREAM_ELEMENT_BYTETRACK_LAPJV_H_
#define SOPHON_STREAM_ELEMENT_BYTETRACK_LAPJV_H_

namespace sophon_stream {
namespace element {
namespace bytetrack {

#define LARGE 1000000

#if !defined TRUE
#define TRUE 1
#endif
#if !defined FALSE
#define FALSE 0
#endif

#define NEW(x, t, n)                            \
  if ((x = (t*)malloc(sizeof(t) * (n))) == 0) { \
    return -1;                                  \
  }
#define FREE(x) \
  if (x != 0) { \
    free(x);    \
    x = 0;      \
  }
#define SWAP_INDICES(a, b) \
  {                        \
    int_t _temp_index = a; \
    a = b;                 \
    b = _temp_index;       \
  }

#if 0
#include <assert.h>
#define ASSERT(cond) assert(cond)
#define PRINTF(fmt, ...) printf(fmt, ##__VA_ARGS__)
#define PRINT_COST_ARRAY(a, n)         \
  while (1) {                          \
    printf(#a " = [");                 \
    if ((n) > 0) {                     \
      printf("%f", (a)[0]);            \
      for (uint_t j = 1; j < n; j++) { \
        printf(", %f", (a)[j]);        \
      }                                \
    }                                  \
    printf("]\n");                     \
    break;                             \
  }
#define PRINT_INDEX_ARRAY(a, n)        \
  while (1) {                          \
    printf(#a " = [");                 \
    if ((n) > 0) {                     \
      printf("%d", (a)[0]);            \
      for (uint_t j = 1; j < n; j++) { \
        printf(", %d", (a)[j]);        \
      }                                \
    }                                  \
    printf("]\n");                     \
    break;                             \
  }
#else
#define ASSERT(cond)
#define PRINTF(fmt, ...)
#define PRINT_COST_ARRAY(a, n)
#define PRINT_INDEX_ARRAY(a, n)
#endif

typedef signed int int_t;
typedef unsigned int uint_t;
typedef double cost_t;
typedef char boolean;
typedef enum fp_t { FP_1 = 1, FP_2 = 2, FP_DYNAMIC = 3 } fp_t;

extern int_t lapjv_internal(const uint_t n, cost_t* cost[], int_t* x, int_t* y);

}  // namespace bytetrack
}  // namespace element
}  // namespace sophon_stream

#endif  // SOPHON_STREAM_ELEMENT_BYTETRACK_LAPJV_H_